# coding = utf-8
import os
import getpass
if getpass.getuser()==0:
    pass
else:
    print("当前用户不是root用户，请以root用户执行脚本")
    exit(1)

version = input('请输入你想安装的python版本（2.7/3.5)')

if version =='2.7':
    url="https://www.python.org/ftp/python/2.7.12/Python-2.7.12.zip"
elif version =='3.5':
    url="https://www.python.org/ftp/python/3.5.2/Python-3.5.2.zip"
else:
    print('您输入的版本号有误，请输入2.7或者3.5')
    exit(1)

cmd = 'wget'+url
res = os.system(cmd)
if res==0:
    print('下载源码失败，请检查网络')
    exit(1)
if version=='2.7':
    package_name ='tar xf Python-2.7.12.tgz'
else:
    package_name ='tar xf Python-3.5.2'
cmd ='tar xf'+package_name+'.tgz'
res=os.system(cmd)
if res!=0:
    os.system('rm'+package_name+'.tgz')
    print('解压源码包失败，请重新运行这个脚本下载源码包')
    exit(1)
cmd='cd '+package_name+'&&./configure --prefix=/usr/local/python && make && make install'
res = os.system(cmd)
if res!=0:
    print('编译python源码失败，请检查是否缺少依赖库')
    exit(1)